tensor([[[0.3400, 0.3400],
         [0.3400, 0.3400]],

        [[0.3700, 0.3700],
         [0.3700, 0.3700]],

        [[0.2900, 0.2900],
         [0.2900, 0.2900]]], grad_fn=<RoundBackward1>)